#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>  // 正确的头文件
#include <unistd.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <sys/wait.h>
#include <string.h>  // 为了使用 strerror
#include <signal.h>
 
#define N 64
 
// 回收进程的资源
void handler(int signum)
{
    wait(NULL);
}
 
int main(int argc, char const *argv[])
{
    int sockfd;
 
    // if (argc < 3)
    // {
    //     printf("usage: %s <ip> <port>\n", argv[0]);
    //     return -1;
    // }
 
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        return -1;
    }
 
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(8080);
    // 自动绑定所有的本机网卡的地址
    addr.sin_addr.s_addr = INADDR_ANY;  // 或者使用 inet_addr(argv[1]) 来绑定特定 IP
 
    int addrlen = sizeof(addr);
 
    if (bind(sockfd, (struct sockaddr *)&addr, addrlen) < 0)
    {
        perror("bind err");
        close(sockfd);
        return -1;
    }
 
    if (listen(sockfd, 5) < 0)
    {
        perror("listen err");
        close(sockfd);
        return -1;
    }
 
    printf("wait client connect\n");
    int clifd;
    struct sockaddr_in cliaddr;
 
    struct sigaction sa;
    memset(&sa, 0, sizeof(sa));
    sa.sa_handler = handler;
    sigaction(SIGCHLD, &sa, NULL);
 
    while (1)
    {
        clifd = accept(sockfd, (struct sockaddr *)&cliaddr, &addrlen);
        if (clifd < 0)
        {
            perror("accept err");
            continue;
        }
 
        pid_t pid = fork();
        if (pid < 0)
        {
            perror("fork err");
            close(clifd);
            continue;
        }
        else if (pid == 0)  // 子进程
        {
            int ret;
            char buf[N] = {0};
            while (1)
            {
                ret = recv(clifd, buf, N, 0);
                if (ret < 0)
                {
                    perror("recv err");
                    continue;
                }
                else if (ret == 0)
                {
                    printf("peer exit\n");
                    break;
                }
                else
                {
                    printf("ip = %s, port = %d, data = %s\n",
                        inet_ntoa(cliaddr.sin_addr), ntohs(cliaddr.sin_port), buf);
                }
            }
            close(clifd);
            exit(0);
        }
        else  // 父进程
        {
            close(clifd);  // 关闭子进程已使用的套接字描述符
        }
    }
 
    close(sockfd);  // 这行代码实际上不会被执行，因为上面的循环是无限循环
 
    return 0;
}